import numpy as np

# 打开文件并读取每一行数据
with open("filename.txt", "r") as file:
    lines = file.readlines()

# 创建一个新的空列表，用于存储处理后的数据
new_data = []

# 遍历每一行数据
for line in lines:
    # 将数据按空格分割
    data = line.split()

    # 将时间戳与后面的数据分开
    timestamp = data[0]
    values = np.array(data[1:], dtype=float)

    # 根据条件进行过滤
    if values[:3].any() != 0 or not (80 <= np.sum(values[3:]**2) <= 100):
        continue

    # 将符合条件的数据添加到新的列表中
    new_data.append(line)

# 将处理后的数据写入新文件
with open("filtered_data.txt", "w") as file:
    file.writelines(new_data)